部署 yolov5 目标检测


1.准备

提前安装好必要程序,以免安装 yolov5 时报错

sudo apt-get update
sudo apt-get upgrade
sudo apt-get install git
sudo apt-get install python3-pip
sudo apt-get install python-is-python3

2.安装 yolov5

项目地址:https://github.com/ultralytics/yolov5

git clone https://github.com/ultralytics/yolov5
cd yolov5
pip install -U -r requirements.txt -i https://pypi.tuna.tsinghua.edu.cn/simple

过程比较慢,请耐心等待


3.运行 yolov5

检测本地图片

默认位置:/yolov5/data/images

输出位置:/yolov5/runs/detect

注:运行 yolov5 时会自动安装一些配置文件

python detect.py

实时检测摄像头中的目标

python detect.py --weights yolov5s.pt --source 0
图片

参数说明:

  • --source 0: 指定摄像头索引(0 表示默认摄像头,若多摄像头可尝试 1, 2 等)
  • --weights yolov5s.pt: 指定模型权重
  • --conf 0.5: 调整置信度阈值(0~1,值越高检测越严格)
  • --img 640: 输入图像分辨率(分辨率越低速度越快)
  • --view-img: 实时显示检测画面(默认不显示,需手动启用)
  • --save-txt: 保存检测结果的标签文件
  • --save-conf: 保存标签文件中的置信度
  • --save-vid: 保存检测后的视频文件
  • --view-img: 可选,实时显示画面(不添加此参数可提升保存速度)
  • --name: 指定保存文件夹名称(避免多次运行覆盖结果)
  • --vid-stride 5: 每 5 帧保存 1 帧(降低视频帧率)
  • --imgsz 480: 输入分辨率(影响输出视频画质和速度)